def add(a, b):
    return a + b


def minus(a, b):
    return a - b


def plus(a, b):
    return a * b


# 测试 a + b 如果函数a + b 等于我们期望的值,那么函数就是有问题的
def test_add(a, b, expect):
    if add(a, b) != expect:
        return False
    else:
        return True


def test_minus(a, b, expect):
    if minus(a, b) != expect:
        return False
    else:
        return True


def test_plus(a, b, expect):
    if plus(a, b) != expect:
        return False
    else:
        return True


if __name__ == "__main__":
    print(test_add(1, 5, 6))

    print(test_minus(5, 5, 1))

    print(test_plus(4, 5, 15))
